import type { AxiosRequestConfig } from 'axios';

// * 声明一个 Map 用于存储每个请求的标识 和 取消函数
let pendingMap = new Map<string, AbortController>();

// * 序列化参数
export const getPendingUrl = (config: AxiosRequestConfig) =>
  [
    config.method,
    config.url,
    JSON.stringify(config.data),
    JSON.stringify(config.params),
  ].join('&');

export class AxiosCanceler {
  /**
   * @description: 添加请求
   * @param {Object} config
   * @return void
   */
  addPending(config: AxiosRequestConfig) {
    // * 在请求开始前，对之前的请求做检查取消操作
    this.removePending(config);
    const url = getPendingUrl(config);
    // 如果该请求已存在，则取消之前的请求
    if (pendingMap.has(url)) {
      const controller = pendingMap.get(url);
      controller?.abort(); // 取消之前的请求
      pendingMap.delete(url); // 移除已取消的控制器
    }

    // 为当前请求创建新的 AbortController
    const controller = new AbortController();
    config.signal = controller.signal; // 将 signal 添加到请求配置中

    // 存储请求标识和控制器的映射
    pendingMap.set(url, controller);
  }

  /**
   * @description: 移除请求
   * @param {Object} config
   */
  removePending(config: AxiosRequestConfig) {
    const url = getPendingUrl(config);

    if (pendingMap.has(url)) {
      // 如果在 pending 中存在当前请求标识，需要取消当前请求，并且移除
      const controller = pendingMap.get(url);
      if (controller) {
        controller.abort();
      }
      pendingMap.delete(url);
    }
  }

  /**
   * @description: 清空所有pending
   */
  removeAllPending() {
    pendingMap.forEach(controller => {
      if (controller) {
        controller.abort();
      }
    });
    pendingMap.clear();
  }

  /**
   * @description: 重置
   */
  reset(): void {
    pendingMap = new Map<string, AbortController>();
  }
}
